#ifndef ORIGINAL_TF_GEN_H
#define	ORIGINAL_TF_GEN_H

#include "TFGenerator.h"
#include "types.h"
#include <math.h>

class MREAgent; 
class OriginalTFGenerator
	:public TFGenerator
{
public:
	OriginalTFGenerator(MREAgent* p, int an, int od, taskspec_t& spec); 
	virtual ~OriginalTFGenerator(); 
	virtual void learn(const Transition* t); 
	virtual Observation predict(Observation st, Action a); 
	virtual double getConfidence(Observation st, Action a); 
	virtual void batchLearn(std::list<Transition>& history); 

private:
	observation_t prev; 
	observation_t next; 
	action_t action; 
};
#endif
